package com.atguigu.srb.core.service.impl;

import com.alibaba.excel.EasyExcel;
import com.atguigu.srb.core.listener.DictExcelVOListener;
import com.atguigu.srb.core.mapper.DictMapper;
import com.atguigu.srb.core.pojo.entity.Dict;
import com.atguigu.srb.core.pojo.vo.DictExcelVO;
import com.atguigu.srb.core.service.DictService;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 数据字典 服务实现类
 * </p>
 *
 * @author Atguigu
 * @since 2024-04-15
 */
@Service
public class DictServiceImpl extends ServiceImpl<DictMapper, Dict> implements DictService {
    @Autowired
    private DictExcelVOListener dictExcelVOListener;
    @Resource
    private DictMapper dictMapper;
    @Autowired
    private RedisTemplate redisTemplate;

    @Cacheable(value = "dict:cache:name",key = "#dictCode+':'+#value")
    @Override
    public String getNameByDictCodeAndValue(String dictCode, Integer value) {
        return dictMapper.getNameByDictCodeAndValue(dictCode,value);
    }

    @Override
    public Boolean importDict(MultipartFile file) {
        try {
            EasyExcel.read(file.getInputStream(), DictExcelVO.class, dictExcelVOListener)
                    .sheet(0)
                    .doRead();
            return true;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return false;
    }

    @Override
    public void exportDict(HttpServletResponse response) {
        try {
            List<Dict> dictList = list();
            // 将一个类型的list转换为另一个类型的list --> stream
            List<DictExcelVO> data = dictList.stream().map(dict -> {
                DictExcelVO dictVO = new DictExcelVO();
                BeanUtils.copyProperties(dict, dictVO);
                return dictVO;
            }).collect(Collectors.toList());
            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
            response.setCharacterEncoding("utf-8");
            // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
            String fileName = URLEncoder.encode("数据字典" + System.currentTimeMillis(), "UTF-8").replaceAll("\\+", "%20");
            response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");

            EasyExcel.write(response.getOutputStream(), DictExcelVO.class)
                    .inMemory(true)
                    .sheet("数据字典")
                    .doWrite(data);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    @Cacheable(value = "srb:dict",key = "#id") // 3. 在业务方法上添加注解  增强缓存功能
    @Override
    public List<Dict> dictsByParentId(Long id) {
        //抽取缓存管理：方式一：自定义注解+aop  电商实现
        //方式二： 简单的缓存管理可以使用springTask提供的自动缓存管理:  不能解决缓存穿透击穿雪崩问题
        //springContext包提供了较多的注解 简化开发：
        //1、缓存管理注解：EnableCaching(启用缓存管理)、Cachable(判断如果没有则设置缓存)、CachePut(保存缓存)
        //                 CacheEvict(删除缓存)
        //步骤：1、启用缓存管理 2、创建缓存管理器(配置缓存方式) 3、在需要缓存的业务方法上添加注解
        //      4、更新数据时可以让缓存失效
        //2、异步多线程注解：EnbleAsync Async(异步执行)
        //根据id查询下一级的数据字典：使用缓存管理

//        手动添加缓存功能
   /*     value+key用来拼接缓存到redis中的key   key中可以获取参数列表拼接key
        List<Dict> dictList = (List<Dict>) redisTemplate.opsForValue().get("srb:dicts:" + id);
        if(dictList != null ){  // 如果缓存中有数据，则直接返回
            return dictList;
        }*/
//         否则查询mysql

        List<Dict> dictList = this.list(Wrappers.lambdaQuery(Dict.class).eq(Dict::getParentId, id));
        dictList.forEach(dict -> {
            // 查看当前节点是否还有下一级子节点  ---> 当前节点的id是否为别的节点的父id
            dict.setHasChildren(hasChildren(dict.getId()));
        });
        // 手动添加缓存功能 查询mysql之后，缓存数据
        // redisTemplate.opsForValue().set("srb:dicts:" + id,dictList);
        return dictList;
    }

    @Override
    public Boolean hasChildren(Long id) {
        // 查看当前节点是否有子节点  ---> 当前节点的id是否为别的节点的父id
        int count = this.count(Wrappers.lambdaQuery(Dict.class).eq(Dict::getParentId, id));
        return count > 0;
    }

    // 当我们新增或者修改数据时，应该让缓存失效，否则缓存的是旧数据，不走数据库
    // 我们添加的是子节点 ---> 所以应该让父节点的缓存失效
    @CacheEvict(value = "srb:dict",key = "#dict.parentId")
    @Override
    public void add(Dict dict) {
        save(dict);
    }

    @Override
    public List<Dict> listByCode(String code) {

        return dictMapper.listByCode(code);

    }

}
